from typing import List

from pydantic import BaseModel


class VectorInsertOrUpdateRequest(BaseModel):
    id: str = ""
    categories: str = ""
    vectorText: str = ""
    originalText: str = ""


class VectorRemoveRequest(BaseModel):
    ids: List[str] = []
    categories: List[str] = []


class VectorQueryRequest(BaseModel):
    text: str
    categories: str
    score_threshold: float
    is_fuzzy: bool = False
